package com.webrtc.config.websocket;

import org.apache.commons.lang3.StringUtils;

import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

/**
 * 连接资源类
 *
 * @author liuzhiqiang
 */
public class WebSocketResources {

    /**
     * userId和websocket得绑定关系
     */
    private static Map<String, WebSocketServerEndpoint> userWebSocketMap = new ConcurrentHashMap<>();


    public static void addWebSocket(String userId, WebSocketServerEndpoint webSocketServerEndpoint) {
        userWebSocketMap.put(userId, webSocketServerEndpoint);
    }

    public static WebSocketServerEndpoint getWebSocketServerEndpoint(String userId) {
        return userWebSocketMap.get(userId);
    }

    public static Map<String, WebSocketServerEndpoint> getUserWebSocketMap() {
        return userWebSocketMap;
    }

    public static void remove(String userId) {
        if (StringUtils.isNotEmpty(userId)) {
            userWebSocketMap.remove(userId);
        }
    }
}
